package programs;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.PriorityBlockingQueue;

public class Test_Timer{

    public static void main(String[] args) {

        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello 睡眠时间到被唤醒的demo");
            }
        },3000);
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello 新插入的demo");
            }
        },500);
        System.out.println("main");

    }
}
// 表示一个任务
class MyTask implements Comparable<MyTask>{
    // 任务具体要干啥
    private Runnable runnable;
    // 任务什么时候干
    private long time;

    // delay 是一个时间间隔,不是绝对的时间
    public MyTask(Runnable runnable,long delay) {
        this.runnable = runnable;
        this.time = System.currentTimeMillis() + delay;
    }

    public void run(){
        runnable.run();
    }
    public long getTime(){
        return this.time;
    }

    @Override
    public int compareTo(MyTask o) {
        return (int)(this.time - o.time);
    }
}
class MyTimer{
    // 能够管理多个任务
    // 用堆,根据时间戳来执行任务
    // 带有阻塞功能的优先级队列 -- 要考虑线程安全问题,可能在多个线程进行注册任务,同时还有线程来执行
    private PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    // 把任务放进队列
    public void schedule(Runnable runnable,long delay){
        MyTask task = new MyTask(runnable, delay);
        queue.put(task);
        synchronized (locker){
            locker.notify();
        }
    }

    private Object locker = new Object();
    // 需要执行最靠前的任务
    // 需要有一个线程来检查 小根堆的顶元素(任务),看是否需要执行了
    public MyTimer(){
        Thread t = new Thread(() -> {
            while (true){
                try {
                    // 取出堆顶任务
                    MyTask task = queue.take();
                    // 看是否到执行时间了
                    long curTime = System.currentTimeMillis();
                    // 如果时间还没到
                    if (curTime < task.getTime()){
                        // 将任务塞回优先级队列
                        queue.put(task);
                        // 等待相应时间 wait()
                        synchronized (locker){
                            locker.wait(task.getTime() - System.currentTimeMillis());
                        }
                    }else {
                        // 时间到了 --- 执行该任务
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        t.start();
    }
}
